package eo.builder.util;

import java.util.HashSet;
import java.util.Set;

public class ClassListingIterationStep extends
        AbstractClassLoadingIterationStep {

    private final Set<Class<?>> classes = new HashSet<Class<?>>();

    public ClassListingIterationStep(ClassLoader classLoader) {
        super(classLoader);
    }

    @Override
    protected boolean classFound(Class<?> clazz) {
        this.classes.add(clazz);
        return true;
    }

    /**
     * 
     * @return {@link Set} of classes found.
     */
    public Set<Class<?>> getClasses() {
        return classes;
    }

}
